package com.atguigu.gulimall.order.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

@Configuration
public class MyRabbitConfig {

    RabbitTemplate rabbitTemplate;

    /**
     * todo 需要优化
     * 自定义rabbitTemplate，因为引入sentinel后，注入的rabbitTemplate会引起循环依赖
     * @param factory
     * @return
     */
    @Primary
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory factory) {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(factory);
        this.rabbitTemplate = rabbitTemplate;
        rabbitTemplate.setMessageConverter(messageConverter());
        initRabbitTemplate();
        return rabbitTemplate;
    }

    /**
     * 使用JSON序列化机制，进行消息转换
     *
     * @return
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     */
    public void initRabbitTemplate() {

        //设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 只要消息抵达Broker（消息代理服务器）ack就等于true
             * @param correlationData 前消息的唯一关联数据（消息的唯一Id）
             * @param ack 消息是否成功收到
             * @param cause 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                /**
                 * 1.做好消息确认机制
                 * 2.每一个发送的消息都在数据库保存好，定期将失败的消息进行重发
                 * todo 消息存放到数据库
                 */
                System.out.println("消息抵达Broker，ack:" + ack + "--cause:" + cause);
            }
        });

        //设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 消息没有抵达队列的失败回调，可以保存到数据库
             * @param message 失败的消息
             * @param replyCode 回复的状态码
             * @param replyText 回复的内容
             * @param exchange 接受消息的交换机
             * @param routingKey 消息使用的路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                //todo 存放到数据库
                System.out.println("消息没有抵达队列，replyCode:" + replyCode + "--replyText:" + replyText + "--exchange:" + exchange + "--routingKey:" + routingKey);
            }
        });
    }
}
